"""
微信AI助手启动脚本
"""
import sys
import os

# 确保当前目录在PATH中
sys.path.insert(0, os.path.abspath("."))

# 将项目根目录添加到Python路径
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))

# 修改导入语句
from wechat.SUD_GetMessage import _init_wechat   # 导入微信实例
from core.logger import get_logger  # 导入日志模块

def main():
    # 如果没有提供参数，给用户一个交互式菜单
    if len(sys.argv) < 2:
        print("请选择启动模式:")
        print("1. 控制台模式 (console)")
        print("2. 界面模式 (ui)")
        
        try:
            choice = input("请输入选项 (1/2): ")
            if choice == '1':
                mode = 'console'
            elif choice == '2':
                mode = 'ui'
            else:
                print("无效的选择")
                return
        except KeyboardInterrupt:
            print("\n已取消")
            return
    else:
        mode = sys.argv[1]
    
    if mode == 'console':
        print("正在启动控制台模式...")
        from app import main
        #main.py 已经有入口点
    elif mode == 'ui':
        print("正在启动界面模式...")
        from app import ui_main
        # ui_main.py已经有入口点
    else:
        print("无效的模式。请使用 'console' 或 'ui'")

if __name__ == "__main__":
    main()
